package com.zjj.algorithm.learning.dmsxl.hash;

import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;

/**
 * 349. 两个数组的交集 简单题
 * 给定两个数组 nums1 和 nums2 ，返回 它们的交集 。输出结果中的每个元素一定是 唯一 的。我们可以 不考虑输出结果的顺序 。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * 输入：nums1 = [1,2,2,1], nums2 = [2,2]
 * 输出：[2]
 * 示例 2：
 * <p>
 * 输入：nums1 = [4,9,5], nums2 = [9,4,9,8,4]
 * 输出：[9,4]
 * 解释：[4,9] 也是可通过的
 * <p>
 * <p>
 * 提示：
 * <p>
 * 1 <= nums1.length, nums2.length <= 1000
 * 0 <= nums1[i], nums2[i] <= 1000
 *
 * @author zjj_admin
 * @date 2022/12/22 14:23
 */
public class Hash_03_TwoArrarIntersection {

    public static void main(String[] args) {
        int[] nums1 = {4, 7, 9, 7, 6, 7};
        int[] nums2 = {5, 0, 0, 6, 1, 6, 2, 2, 4};
        int[] res = intersection2(nums1, nums2);
        System.out.println("res = " + Arrays.toString(res));
    }


    /**
     * 时间
     * 4 ms
     * 击败
     * 38.21%
     * 内存
     * 42.1 MB
     * 击败
     * 5.2%
     *
     * @param nums1
     * @param nums2
     * @return
     */
    public static int[] intersection2(int[] nums1, int[] nums2) {
        int[] arr1 = new int[1000];
        for (int i : nums1) {
            arr1[i] = 1;
        }
        for (int num : nums2) {
            if (arr1[num] != 0) {
                arr1[num] = 2;
            }
        }
        Set<Integer> res = new HashSet<>();

        for (int i = 0; i < arr1.length; i++) {
            if (arr1[i] == 2) {
                res.add(i);
            }
        }
        return res.stream().mapToInt(x -> x).toArray();
    }

    /**
     * 时间
     * 4 ms
     * 击败
     * 38.21%
     * 内存
     * 41.4 MB
     * 击败
     * 64.74%
     *
     * @param nums1
     * @param nums2
     * @return
     */
    public static int[] intersection1(int[] nums1, int[] nums2) {
        Set<Integer> set1 = new HashSet<>();
        Set<Integer> set2 = new HashSet<>();
        for (int num : nums1) {
            set1.add(num);
        }
        for (int num : nums2) {
            if (set1.contains(num)) {
                set2.add(num);
            }
        }
        return set2.stream().mapToInt(x -> x).toArray();
    }
}
